from pydantic import BaseModel


class TestRequest(BaseModel):
    system: str
    user: str


class TestResponse(BaseModel):
    answer: str
